import os
from langchain_core.messages import HumanMessage, SystemMessage
from langchain_openai import ChatOpenAI
from langchain.chains import LLMChain
from langchain_core.prompts import PromptTemplate
from llm_base import llm




template = """问题: {question}

答案: 请一步一步的思考。"""

prompt_template = PromptTemplate.from_template(template)
chain = LLMChain(prompt=prompt_template, llm=llm)

response = chain.invoke("什么是快乐星球")

print(response)

